Notebook de classification du modèle entraîné avec photos rognées

Adrian Rodriguez

Projet 7 parcours Ingénieur Machine Learning

Accès projet git

Plus d'informations


0. Contexte

Un rapport sur les erreurs de classification des races de chiens a été soumis à l’association de protection des animaux. Dans ce rapport, il est recommandé d’implémenter une solution de détection d’objet pour améliorer la qualité des photos. Cette solution doit isoler l’objet à classifier en procédant à un rognage. Cette nouvelle méthode fera appel au dernier YOLO, version 4.

Sources bibliographiques :

Prototype

Ce notebook présente les résultats de classification du modèle Xception entraînés avec les photos rognées.


1. Chargement du modèle et des fichiers nécessaires


2. Extraction des prédictions en erreur


3. Etude des prédictions en erreur

3-1. Visualisation d'un échantillon d'erreur

Les erreurs sont désormais dues à des erreurs d'attribution des races, à des photos de jeunes chiens, ou à des races qui se ressemblent fortement.


4. Conclusion

Le modèle entraîné sur les photos rognés donne de meilleurs performance (+4 %). Recommendations :